from langchain_core.messages import SystemMessage, HumanMessage
from langchain_core.output_parsers import StrOutputParser

from base import model

# 创建大语言模型
# 输入提示词模板
prompt = [
    SystemMessage(content="你是一个用于Java开发的智能代码助手"),
    HumanMessage(content="写一个快速排序方法")
]
print(prompt)
# chain = prompt.format({"input":"写一个快速排序算法"}) | llm

# 执行大模型计算 获取 结果
modelResult = model.invoke(prompt)
print(f"-----model-result----- \n${modelResult}")

# 大模型输出解析
parseResult =StrOutputParser().invoke(modelResult)
print(f"-----parsed result-----\n${parseResult}")
